from cgitb import small
import os
from PIL import Image
import numpy as np
def getSmallImages(images_directory):
image_files = os.listdir(images_directory)
images = []
for file in image_files:
filepath = os.path.join(images_directory,file)
try:
img = Image.open(filepath).convert('RGB')
images.append(img)
except:
print("Invalid Small Image:{}".format(filepath))
return images
def processImages(images, method = 'RESIZE'):
processed_images = []
if(method == 'CROP'):
for img in images:
w,h = img.size
l = min(w,h)
left = (w-l)/2
right = w - left
top = (h-l)/2
bottom = h - top
processed = img.crop((left,top,right,bottom))
processed_images.append(processed)
elif(method == 'RESIZE'):
for img in images:
l = min(img.size[:])
processed = img.resize((l,l))
processed_images.append(processed)
else:
print("ERR! PROCESS METHOD NOT SUPPORTED!")
return processed_images
# small_images = getSmallImages('./pics/smalls/')
small_images = getSmallImages('./pics/phone/')
# smallImages = processImages(smallImages, method = 'CROP')
# small_images = processImages(small_images, method = 'RESIZE')
small_images[0]